#!/usr/bin/env python
# -*- coding: utf-8 -*-
#   
#       Copyright 2010 Santiago Lezica, Juan Pablo Orsay, Facundo Martinez.
#       
#       This program is free software; you can redistribute it and/or modify
#       it under the terms of the GNU General Public License as published by
#       the Free Software Foundation; either version 2 of the License, or
#       (at your option) any later version.
#       
#       This program is distributed in the hope that it will be useful,
#       but WITHOUT ANY WARRANTY; without even the implied warranty of
#       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#       GNU General Public License for more details.
#       
#       You should have received a copy of the GNU General Public License
#       along with this program; if not, write to the Free Software
#       Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#       MA 02110-1301, USA.

import types

class ModuleScanner:

    @staticmethod
    def getSubclassesOf(module, klass, ommit = []):
        ret = []

        if type(module) in types.StringTypes:
            module = __import__(module, fromlist = "*") 

        if type(module) is types.ModuleType:

            ommit.append(module)
            
            for obj in [getattr(module, obj_name) for obj_name in dir(module)]:
                if hasattr(obj, "__name__") and not obj.__name__.startswith("_"):

                    if type(obj) is types.ModuleType and obj not in ommit:
                        # recursive call.
                        ret += ModuleScanner.getSubclassesOf(obj, klass, ommit)

                    else:
                        try:
                            if issubclass(obj, klass) and obj is not klass:
                                ret.append(obj)
                        except:
                            pass

        else:
            raise TypeError("First parameter should be module, or module name")
            
        return ret
